package com.zjsru.plan2023.oneday;

/**
 * @Author: cookLee
 * @Date: 2023-10-05
 * 买卖股票的最佳时机含冷冻期
 */
public class MaxProfit {

    /**
     * 主
     * \
     * 输入: prices = [1,2,3,0,2]
     * 输出: 3
     * 解释: 对应的交易状态为: [买入, 卖出, 冷冻期, 买入, 卖出]
     * \
     * 输入: prices = [1]
     * 输出: 0
     * \
     * @param args args
     */
    public static void main(String[] args) {
        MaxProfit maxProfit = new MaxProfit();
        int[] prices = new int[]{1, 2, 3, 0, 2};
        System.out.println(maxProfit.maxProfit(prices));
    }

    /**
     * 最大利润
     *
     * @param prices 价格
     * @return int
     */
    public int maxProfit(int[] prices) {
        int len = prices.length;
        if (len == 0) {
            return 0;
        }
        int[][] store = new int[len][3];
        //初始化股票,默认买入第一个
        store[0][0] = -prices[0];
        // f[i][0]: 手上持有股票的最大收益
        // f[i][1]: 手上不持有股票，并且处于冷冻期中的累计最大收益
        // f[i][2]: 手上不持有股票，并且不在冷冻期中的累计最大收益
        for (int i = 1; i < len; i++) {
            store[i][0] = Math.max(store[i - 1][0], store[i - 1][2] - prices[i]);
            store[i][1] = store[i - 1][0] + prices[i];
            store[i][2] = Math.max(store[i - 1][1], store[i - 1][2]);
        }
        return Math.max(store[len - 1][1], store[len - 1][2]);
    }

}
